"""
@-*- coding: utf-8 -*-
@ python: 3.12.3
@ 创建者: JacksonCode
@ 创建时间: 2024/12/21
"""
import numpy as np

array = np.array([[1, 5, 8, 4, 3, 9], [6, 5, 4, 9, 3, 0]])

print(array)

print(np.sort(array))
print(np.sort(array, axis=0))

array = np.array([[1.4, 5.1, 7.6, 4.6, 7.8], [3.2, 4.9, 8.1, 1.3, 5.2]])
print(np.argsort(array))  # 给原来元素位置的索引

array2 = np.linspace(0, 10, 10)
print(array2)

values = np.array([2.5, 6.5, 9.6])
print(np.searchsorted(array2, values))  # 想在什么位置插入什么值，   通过这函数查看一下索引   [3 6 9]  (排序好的)

array3 = np.array([[1, 0, 6], [2, 9, 3], [5, 8, 7], [0, 6, 7]])
print(array3)

# 分开排序
index = np.lexsort([-1 * array3[:, 0], array3[:, 2]])
print(index)  # [1 0 2 3]

print(array3[index])
